package algorithm7.leetcode.每日一题.YEAR2023.month6;

import org.junit.jupiter.api.Test;

public class DAY20230602_LC2559 {

    public int[] vowelStrings(String[] words, int[][] queries) {
//        return handle1(words, queries);
        return handle2(words, queries);
    }

    /**
     *  前缀和
     */
    private int[] handle2(String[] words, int[][] queries) {
        int n = words.length;
        int[] cnt = new int[n + 1];
        for (int i = 0; i < n; i++) {
            String word = words[i];
            if ((word.startsWith("a") || word.startsWith("e") || word.startsWith("i") || word.startsWith("o") || word.startsWith("u")) &&
                    (word.endsWith("a") || word.endsWith("e") || word.endsWith("i") || word.endsWith("o") || word.endsWith("u"))) {
                cnt[i + 1] = cnt[i] + 1;
                continue;
            }
            cnt[i + 1] = cnt[i];
        }
        int[] res = new int[queries.length];
        for (int i = 0; i < queries.length; i++) {
            int start = queries[i][0], end = queries[i][1];
            res[i] = cnt[end + 1] - cnt[start];
        }
        return res;
    }

    /**
     * 最后一个用例超时
     */
    private int[] handle1(String[] words, int[][] queries) {
        int n = words.length;
        int[] cnt = new int[n];
        for (int i = 0; i < n; i++) {
            String word = words[i];
            if ((word.startsWith("a") || word.startsWith("e") || word.startsWith("i") || word.startsWith("o") || word.startsWith("u")) &&
                    (word.endsWith("a") || word.endsWith("e") || word.endsWith("i") || word.endsWith("o") || word.endsWith("u"))) {
                cnt[i] = 1;
            }
        }
        int[] res = new int[queries.length];
        for (int i = 0; i < queries.length; i++) {
            int start = queries[i][0];
            int end = queries[i][1];
            for (int j = start ; j <= end ; j++) {
                res[i] += cnt[j];
            }
        }
        return res;
    }

    /*

2559. 统计范围内的元音字符串数

给你一个下标从 0 开始的字符串数组 words 以及一个二维整数数组 queries 。
每个查询 queries[i] = [li, ri] 会要求我们统计在 words 中下标在 li 到 ri 范围内（包含 这两个值）并且以元音开头和结尾的字符串的数目。
返回一个整数数组，其中数组的第 i 个元素对应第 i 个查询的答案。
注意：元音字母是 'a'、'e'、'i'、'o' 和 'u' 。

示例 1：
输入：words = ["aba","bcb","ece","aa","e"], queries = [[0,2],[1,4],[1,1]]
输出：[2,3,0]
解释：以元音开头和结尾的字符串是 "aba"、"ece"、"aa" 和 "e" 。
查询 [0,2] 结果为 2（字符串 "aba" 和 "ece"）。
查询 [1,4] 结果为 3（字符串 "ece"、"aa"、"e"）。
查询 [1,1] 结果为 0 。
返回结果 [2,3,0] 。

     */

    @Test
    void t() {
        String[] words = {"aba", "bcb", "ece", "aa", "e"};
        int[][] queries = {{0, 2}, {1, 4}, {1, 1}};
        int[] res = vowelStrings(words, queries);
        for (int r : res) {
            System.out.print(r + " ");
        }
    }

}
